package com.demo.test;

import java.io.FileInputStream;
import java.io.IOException;
import java.util.HashMap;

public class GetFileType {
	public static void main(String[] args) throws IOException {
		final String fileType = getFileType("C:/Users/Fairylan/Desktop/out");
		System.out.println(fileType);

	}

	// 缓存文件头信息-文件头信息
	public static final HashMap<String, String> FILE_TYPE_MAP = new HashMap<String, String>();
	static {
		FILE_TYPE_MAP.put("FFD8FF", "jpg"); // JPEG (jpg)
		FILE_TYPE_MAP.put("89504E47", "png"); // PNG (png)
		FILE_TYPE_MAP.put("47494638", "gif"); // GIF (gif)
		FILE_TYPE_MAP.put("49492A00", "tif"); // TIFF (tif)
		FILE_TYPE_MAP.put("424D", "bmp"); // Windows Bitmap (bmp)
		FILE_TYPE_MAP.put("41433130", "dwg"); // CAD (dwg)
		FILE_TYPE_MAP.put("68746D6C3E", "html"); // HTML (html)
		FILE_TYPE_MAP.put("7B5C727466", "rtf"); // Rich Text Format (rtf)
		FILE_TYPE_MAP.put("3C3F786D6C", "xml");
		FILE_TYPE_MAP.put("504B0304", "zip");
		FILE_TYPE_MAP.put("52617221", "rar");
		FILE_TYPE_MAP.put("38425053", "psd"); // Photoshop (psd)
		FILE_TYPE_MAP.put("44656C69766572792D646174653A", "eml"); // Email [thorough only] (eml)
		FILE_TYPE_MAP.put("CFAD12FEC5FD746F", "dbx"); // Outlook Express (dbx)
		FILE_TYPE_MAP.put("2142444E", "pst"); // Outlook (pst)
		FILE_TYPE_MAP.put("D0CF11E0", "xls"); // MS Word
		FILE_TYPE_MAP.put("D0CF11E0", "doc"); // MS Excel 注意：word 和 excel的文件头一样
		FILE_TYPE_MAP.put("5374616E64617264204A", "mdb"); // MS Access (mdb)
		FILE_TYPE_MAP.put("FF575043", "wpd"); // WordPerfect (wpd)
		FILE_TYPE_MAP.put("252150532D41646F6265", "eps");
		FILE_TYPE_MAP.put("252150532D41646F6265", "ps");
		FILE_TYPE_MAP.put("255044462D312E", "pdf"); // Adobe Acrobat (pdf)
		FILE_TYPE_MAP.put("AC9EBD8F", "qdf"); // Quicken (qdf)
		FILE_TYPE_MAP.put("E3828596", "pwl"); // Windows Password (pwl)
		FILE_TYPE_MAP.put("57415645", "wav"); // Wave (wav)
		FILE_TYPE_MAP.put("41564920", "avi");
		FILE_TYPE_MAP.put("2E7261FD", "ram"); // Real Audio (ram)
		FILE_TYPE_MAP.put("2E524D46", "rm"); // Real Media (rm)
		FILE_TYPE_MAP.put("000001BA", "mpg"); //
		FILE_TYPE_MAP.put("6D6F6F76", "mov"); // Quicktime (mov)
		FILE_TYPE_MAP.put("3026B2758E66CF11", "asf"); // Windows Media (asf)
		FILE_TYPE_MAP.put("4D546864", "mid"); // MIDI (mid)
	}

	/**
	 * 方法描述：根据文件路径获取文件头信息
	 * @param filePath 文件路径
	 * @return 文件头信息
	 */
	public static String getFileType(String filePath) {
		return FILE_TYPE_MAP.get(getFileHeader(filePath));
	}

	/**
	 * 方法描述：根据文件路径获取文件头信息
	 * @param filePath 文件路径
	 * @return 文件头信息
	 */
	public static String getFileHeader(String filePath) {
		FileInputStream is = null;
		String value = null;
		try {
			is = new FileInputStream(filePath);
			byte[] b = new byte[4];
			/*
			 * int read() 从此输入流中读取一个数据字节。int read(byte[] b) 从此输入流中将最多 b.length 个字节的数据读入一个
			 * byte 数组中。 int read(byte[] b, int off, int len) 从此输入流中将最多 len 个字节的数据读入一个 byte
			 * 数组中。
			 */
			is.read(b, 0, b.length);
			value = bytesToHexString(b);
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			if (null != is) {
				try {
					is.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
		return value;
	}

	/**
	 * 方法描述：将要读取文件头信息的文件的byte数组转换成string类型表示
	 * @param src 要读取文件头信息的文件的byte数组
	 * @return 文件头信息
	 */
	private static String bytesToHexString(byte[] src) {
		StringBuilder builder = new StringBuilder();
		if (src == null || src.length <= 0) {
			return null;
		}
		String hv;
		for (int i = 0; i < src.length; i++) {
			// 以十六进制（基数 16）无符号整数形式返回一个整数参数的字符串表示形式，并转换为大写
			hv = Integer.toHexString(src[i] & 0xFF).toUpperCase();
			if (hv.length() < 2) {
				builder.append(0);
			}
			builder.append(hv);
		}
		// System.out.println(builder.toString());
		return builder.toString();
	}

}
